package com.dzu.easy;

import java.util.HashSet;
import java.util.Set;

/**
 * @author ZhaoDong
 * @date 2022/8/4 10:16
 * @description leetcode
 * 202. 快乐数
 * 编写一个算法来判断一个数 n 是不是快乐数。
 *
 * 「快乐数」 定义为：
 *
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：n = 19
 * 输出：true
 * 解释：
 * 1^2 + 9^2 = 82
 * 8^2 + 2^2 = 68
 * 6^2 + 8^2 = 100
 * 1^2 + 0^2 + 0^2 = 1
 * 示例 2：
 *
 * 输入：n = 2
 * 输出：false
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/happy-number
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 */
public class Test202 {

    public static void main(String[] args) {

        boolean happy = isHappy(19);
        boolean happy2 = isHappy2(2);
        System.out.println(happy2);
    }

    /**
     * 快慢指针
     * @param n
     * @return
     */
    public static boolean isHappy2(int n) {
        int slow = n, fast = n;
        do{
            slow = getNext2(slow);
            fast = getNext2(fast);
            fast = getNext2(fast);
        }while(slow != fast);

        return slow == 1;
    }
    private static int getNext2(int n) {
        int sum = 0;
        while(n > 0)
        {
            int bit = n % 10;
            sum += bit * bit;
            n = n / 10;
        }
        return sum;
    }


    /**
     * 利用集合
     * @param n
     * @return
     */
    public static boolean isHappy(int n) {
        Set<Integer> seen = new HashSet<>();
        while (n != 1 && !seen.contains(n)) {
            seen.add(n);
            n = getNext(n);
        }
        return n == 1;

    }
    private static int getNext(int n) {
        int sum = 0;
        while (n > 0) {
            int d = n % 10;
            n = n / 10;
            sum += d * d;
        }
        return sum;
    }

}
